#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
启动基于向量搜索的树形结构Neo4j查询API服务
"""

import uvicorn
import sys
import os

def main():
    """启动API服务"""
    print("🚀 启动基于向量搜索的树形结构Neo4j查询API服务")
    print("=" * 60)
    print("📋 服务信息:")
    print("   - 服务名称: 向量搜索树形查询API")
    print("   - 版本: 2.0.0")
    print("   - 端口: 8001")
    print("   - 文档地址: http://localhost:8001/docs")
    print("   - 健康检查: http://localhost:8001/health")
    print("=" * 60)
    
    # 检查依赖
    try:
        from vector_tree_query_tool import VectorTreeQueryTool
        print("✅ 向量搜索工具模块加载成功")
    except ImportError as e:
        print(f"❌ 无法导入向量搜索工具: {e}")
        return
    
    try:
        # 启动服务
        uvicorn.run(
            "vector_tree_query_api:app",
            host="0.0.0.0",
            port=8001,
            reload=True,
            log_level="info"
        )
    except KeyboardInterrupt:
        print("\n👋 服务已停止")
    except Exception as e:
        print(f"❌ 服务启动失败: {e}")

if __name__ == "__main__":
    main()
